package com.linyaonan.leetcode.easy._888;

import java.util.HashSet;

/**
 *
 * 返回一个整数数组 ans，其中 ans[0] 是爱丽丝必须交换的糖果棒的大小，ans[1] 是 Bob 必须交换的糖果棒的大小。
 *
 * 如果有多个答案，你可以返回其中任何一个。保证答案存在。
 *
 *  
 *
 * 示例 1：
 *
 * 输入：A = [1,1], B = [2,2]
 * 输出：[1,2]
 * 示例 2：
 *
 * 输入：A = [1,2], B = [2,3]
 * 输出：[1,2]
 * 示例 3：
 *
 * 输入：A = [2], B = [1,3]
 * 输出：[2,3]
 * 示例 4：
 *
 * 输入：A = [1,2,5], B = [2,4]
 * 输出：[5,4]
 *  
 *
 * 提示：
 *
 * 1 <= A.length <= 10000
 * 1 <= B.length <= 10000
 * 1 <= A[i] <= 100000
 * 1 <= B[i] <= 100000
 * 保证爱丽丝与鲍勃的糖果总量不同。
 * 答案肯定存在。
 *
 *
 * @author: Lin
 * @date: 2020/1/13
 */
public class FairCandySwap {

    public int[] fairCandySwap(int[] A, int[] B) {
        long aCount = 0;
        long bCount = 0;
        for (int i : A) {
            aCount += i;
        }
        HashSet<Integer> set = new HashSet<>();
        for (int i : B) {
            bCount += i;
            set.add(i);
        }
        int c = (int) ((aCount - bCount) / 2);
        for (int i : A) {
            if (set.contains(i - c)) {
                return new int[]{i, i - c};
            }
        }
        return null;
    }
}
